tensor([[[0.4700, 0.4700],
         [0.4700, 0.4700]],

        [[0.2600, 0.2600],
         [0.2600, 0.2600]],

        [[0.2700, 0.2700],
         [0.2700, 0.2700]]], grad_fn=<RoundBackward1>)